package com.meiyu.live.util;

import java.security.Key;
import java.security.SecureRandom;
import java.security.spec.AlgorithmParameterSpec;

import javax.crypto.Cipher;
import javax.crypto.SecretKey;
import javax.crypto.SecretKeyFactory;
import javax.crypto.spec.DESKeySpec;
import javax.crypto.spec.IvParameterSpec;

/**
 * Created by LI on 2016/8/4.
 */
public class DesUtil {

    public static final String ALGORITHM_DES = "DES/ECB/PKCS5Padding";

    public static byte[] encode(String src, String password) throws Exception {
        try {
            byte[] encryptedData = null;
            try {
                // DES算法要求有一个可信任的随机数源
                SecureRandom sr = new SecureRandom();
                DESKeySpec deskey = new DESKeySpec(password.getBytes("UTF-8"));
                // 创建一个密匙工厂，然后用它把DESKeySpec转换成一个SecretKey对象
                SecretKeyFactory keyFactory = SecretKeyFactory.getInstance("DES");
                SecretKey key = keyFactory.generateSecret(deskey);
                // 加密对象
                Cipher cipher = Cipher.getInstance(ALGORITHM_DES);
                cipher.init(Cipher.ENCRYPT_MODE, key);
                // 加密，并把字节数组编码成字符串
                encryptedData = cipher.doFinal(src.getBytes("UTF-8"));
            } catch (Exception e) {
//            log.error("加密错误，错误信息：", e);
                throw new RuntimeException("加密错误，错误信息：", e);
            }
            return encryptedData;
        } catch (Throwable e) {
            e.printStackTrace();
        }
        return null;
    }


    /**
     * DES算法，解密
     *
     * @param data 待解密字符串
     * @param key  解密私钥，长度不能够小于8位
     * @return 解密后的字节数组
     * @throws Exception 异常
     */
    public static String decode(String key, String data) {
        if (data == null)
            return null;
        try {
            DESKeySpec dks = new DESKeySpec(key.getBytes());
            SecretKeyFactory keyFactory = SecretKeyFactory.getInstance("DES");
            //key的长度不能够小于8位字节
            Key secretKey = keyFactory.generateSecret(dks);
            Cipher cipher = Cipher.getInstance(ALGORITHM_DES);
            IvParameterSpec iv = new IvParameterSpec("12345678".getBytes());
            AlgorithmParameterSpec paramSpec = iv;
            cipher.init(Cipher.DECRYPT_MODE, secretKey, paramSpec);
            return new String(cipher.doFinal(hex2byte(data.getBytes())));
        } catch (Exception e) {
            e.printStackTrace();
            return data;
        }
    }

    /**
     * 二行制转字符串
     *
     * @param b
     * @return
     */
    public static String byte2hex(byte[] b) {
        StringBuilder hs = new StringBuilder();
        String stmp;
        for (int n = 0; b != null && n < b.length; n++) {
            stmp = Integer.toHexString(b[n] & 0XFF);
            if (stmp.length() == 1)
                hs.append('0');
            hs.append(stmp);
        }
        return hs.toString().toUpperCase();
    }

    private static byte[] hex2byte(byte[] b) {
        if ((b.length % 2) != 0)
            throw new IllegalArgumentException();
        byte[] b2 = new byte[b.length / 2];
        for (int n = 0; n < b.length; n += 2) {
            String item = new String(b, n, 2);
            b2[n / 2] = (byte) Integer.parseInt(item, 16);
        }
        return b2;
    }

}